-
Notifications
You must be signed in to change notification settings - Fork 191
Run CI on Modal, upgrade Bitsandbytes #641
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #641 +/- ##
==========================================
- Coverage 85.39% 85.20% -0.20%
==========================================
Files 81 81
Lines 8006 8049 +43
==========================================
+ Hits 6837 6858 +21
- Misses 1169 1191 +22
🚀 New features to boost your workflow:
|
samples_with_tasks = {sample_idx for sample_idx, _ in task_to_indices.values()} | ||
pending_samples = len(samples_with_tasks) # samples for which we have less than k_min results | ||
assert pending_samples <= num_samples |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's better to recompute the number of pending_samples, as some samples might have 0 tasks (for example, in test_call_many
)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Appreciate the version updates and a face lift to the tests. LGTM; pending @dvmazur
* Run CI on Modal, upgrade Bitsandbytes * Extract the blocksize for quantization into a constant (cherry picked from commit 767afa5)
This PR switches the execution of tests from GitHub Actions to Modal, unlocking the option to use GPUs in those tests in the future. Since Modal workers can have multiple CPUs, we can also run tests in parallel, which speeds them up by approximately 4x — from 8 minutes to under 2 minutes.
Also, as multiple tests seem to be unstable, this PR also disables them for the time being or marks them as flaky with
pytest.mark.xfail
. This will be fixed in future PRs.Lastly, since the current version of Bitsandbytes is outdated, the PR upgrades it